# 需求1：把 aa 文件夹所有文件重命名 -- 文件名前增加 "Python_"
# 需求2：把 aa 文件夹所有文件重命名 -- 文件名前删除 "Python_"

import os

# 1、找到所有文件：获取aa文件夹的目录列表 -- listdir()
file_list = os.listdir('aa')

os.chdir('aa')  # 改变操作的目录

# 1表示给文件名加前缀，2表示删除文件名的前缀
print('请输入您要对文件名执行的操作：')
print('1、给文件名加前缀"Python_"')
print('2、删除文件名的前缀"Python_"')
flag = input()

# 2、构造名字
for name in file_list:
    if flag == '1':
        new_name = 'Python_' + name

    elif flag == '2':
        num = len('Python_')
        new_name = name[num:]

    # 3、重命名
    os.rename(name, new_name)


